import os


def rename_jpg_files(folder_path, start_index=1):
    """
    重命名指定文件夹中的所有.jpg文件，按照000001.jpg, 000002.jpg的格式进行编号重命名。

    参数:
    folder_path -- 包含.jpg文件的文件夹路径
    start_index -- 重命名的起始编号，默认从1开始
    """
    # 确保文件夹路径存在
    if not os.path.isdir(folder_path):
        print(f"The specified folder does not exist: {folder_path}")
        return

    # 初始化计数器
    counter = start_index
    # 遍历文件夹中的所有文件
    for filename in os.listdir(folder_path):
        # 检查文件扩展名是否为.jpg
        if filename.lower().endswith('.jpg'):
            # 构造新的文件名
            new_filename = f"{str(counter).zfill(6)}.jpg"
            # 构造完整的文件路径
            old_file_path = os.path.join(folder_path, filename)
            new_file_path = os.path.join(folder_path, new_filename)
            # 检查新文件名是否已存在
            if not os.path.exists(new_file_path):
                # 重命名文件
                os.rename(old_file_path, new_file_path)
                print(f"Renamed: {filename} -> {new_filename}")
            else:
                print(f"Skipping: {filename}, as {new_filename} already exists.")
            # 更新计数器
            counter += 1


# 指定要处理的文件夹路径
folder_path = 'VOCdevkit/VOC2007/JPEGImages'  # 请替换为你的.jpg文件所在的文件夹路径
rename_jpg_files(folder_path)